import os
from pathlib import Path
from docx2pdf import convert

from 政务网.生成工单 import 清空文件夹

# 获取当前脚本所在目录
current_script_dir = os.path.dirname(os.path.abspath(__file__))

# 构建输入和输出目录路径（修正了路径分隔符）
docx_dir = Path(current_script_dir) / "工单填写"
pdf_dir = Path(current_script_dir) / "工单PDF"

print(f"📂 DOCX目录: {docx_dir}")
print(f"📂 PDF目录: {pdf_dir}")

# 创建输出目录（如果不存在）
pdf_dir.mkdir(parents=True, exist_ok=True)

# 检查输入目录是否存在
if not docx_dir.exists():
    print(f"❌ 错误：目录不存在 - {docx_dir}")
    exit(1)

# 检查是否有DOCX文件
docx_files = list(docx_dir.glob("*.docx"))
if not docx_files:
    print(f"❌ 错误：在 {docx_dir} 中未找到DOCX文件")
    exit(1)

print(f"✅ 找到 {len(docx_files)} 个DOCX文件")

try:
    # 批量转换（修正了输出目录参数）
    convert(str(docx_dir), str(pdf_dir))
    print(f"🎉 转换完成！PDF文件保存在: {pdf_dir}")

except Exception as e:
    print(f"❌ 转换失败: {e}")
